/*
 * Copyright 2023 devaper
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.github.devaper.pigeon.handler.argument

import io.github.devaper.pigeon.annotation.ah.*
import io.github.devaper.pigeon.handler.method.*
import io.github.devaper.pigeon.request.*

class HeaderArgumentHandler : AbstractMultiValueArgumentHandler() {

    override fun determineName(methodHandler: AbstractHttpMethodHandler, parameterIndex: Int): String {
        val methodContext = methodHandler.methodContext
        val parameter = methodContext.parameters[parameterIndex]
        return parameter.getAnnotation(Header::class.java)?.let {
            it.value.takeIf { it.isNotEmpty() }
        } ?: parameter.name
    }

    override fun addValue(requestTemplate: RequestTemplate, name: String, value: String) {
        requestTemplate.addHeader(name, value)
    }
}